package per.hx.bs.base.domain;

import java.util.HashMap;
import java.util.Map;

/**
 * @author HuXing
 * @date 2018/1/29  19:11
 * <p>
 * Describe:    返回的 数据模型
 */
public class ReturnDTO extends HashMap<String, Object> {

    private ReturnDTO() {
        put("code", 0);
        put("msg", "操作成功");
    }

    public ReturnDTO(int code, String msg) {
        put("code", code);
        put("msg", msg);
    }

    public ReturnDTO(Integer code, Object data) {
        put("code", code);
        put("msg", "操作成功");
        put("data", data);
    }

    public ReturnDTO(Integer code, String msg, Object data) {
        put("code", code);
        put("msg", msg);
        put("data", data);
    }

    public static ReturnDTO error() {
        return error(1, "操作失败");
    }

    public static ReturnDTO error(String msg) {
        return error(500, msg);
    }

    public static ReturnDTO error(int code, String msg) {
        ReturnDTO returnDTO = new ReturnDTO();
        returnDTO.put("code", code);
        returnDTO.put("msg", msg);
        return returnDTO;
    }

    public static ReturnDTO error(String msg, Object data) {
        ReturnDTO returnDTO = ReturnDTO.error();
        returnDTO.put("msg", msg);
        returnDTO.put("data", data);
        return returnDTO;
    }

    public static ReturnDTO ok() {
        return new ReturnDTO();
    }


    public static ReturnDTO ok(String msg) {
        ReturnDTO returnDTO = new ReturnDTO();
        returnDTO.put("msg", msg);
        return returnDTO;
    }

    public static ReturnDTO ok(String msg, Object data) {
        ReturnDTO returnDTO = new ReturnDTO();
        returnDTO.put("msg", msg);
        returnDTO.put("data", data);
        return returnDTO;
    }

    public static ReturnDTO ok(Map<String, Object> map) {
        ReturnDTO returnDTO = new ReturnDTO();
        returnDTO.putAll(map);
        return returnDTO;
    }

    public static ReturnDTO ok(Object data) {
        ReturnDTO returnDTO = new ReturnDTO();
        returnDTO.put("data", data);
        return returnDTO;
    }


    public static ReturnDTO status(Integer code, String msg, Object data) {
        return new ReturnDTO(code, msg, data);
    }

    public static ReturnDTO check(Integer i) {
        if (i != null && i > 0) {
            return ok();
        }
        return error();
    }

    public static ReturnDTO check(Integer i, String ok) {
        if (i > 0) {
            return ok(ok);
        }
        return error();
    }

    public static ReturnDTO check(Integer i, String ok, String error) {
        if (i > 0) {
            return ok(ok);
        }
        return error(error);
    }

    @Override
    public ReturnDTO put(String key, Object value) {
        super.put(key, value);
        return this;
    }

    @Override
    public String toString() {
        return "ReturnDTO{}";
    }
}
